package com.itcam.cammygateway;

import com.itcam.iahclientsdk.utils.SignUtils;
import com.itcam.project.entity.InterfaceInfo;
import com.itcam.cammycommon.service.InnerInterfaceInfoService;
import com.itcam.cammycommon.service.InnerUserInterfaceInfoService;
import com.itcam.cammycommon.service.InnerUserService;
import lombok.extern.slf4j.Slf4j;
import org.apache.dubbo.config.annotation.DubboReference;
import org.reactivestreams.Publisher;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.core.io.buffer.DataBufferFactory;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.core.io.buffer.DefaultDataBufferFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.http.server.reactive.ServerHttpResponseDecorator;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * description:全局过滤
 */
@Slf4j
@Component
public class CustomGlobalFilter implements GlobalFilter, Ordered {

    @DubboReference
    private InnerUserService innerUserService;

    @DubboReference
    private InnerInterfaceInfoService innerInterfaceInfoService;

    @DubboReference
    private InnerUserInterfaceInfoService innerUserInterfaceInfoService;

    private static final List<String> IP_WHITE_LIST = Arrays.asList("127.0.0.1");

    private static final String INTERFACE_HOST = "http://localhost:8123";

    /**
     * exchange(路由交换机)：我们所有的请求的信息、响应的信息、响应体、请求体都能从这里拿到
     * chain（责任链模式）：因为我们的所有过滤器是按照从上到下的顺序依次执行，形成一个链条。所以这里用了一个chain，如果当前过滤器
     *      对请求进行了过滤后发现可以放行，就要调用责任链中的next方法，相当于直接找到下一个过滤器，这里称为filter。有时候
     *      我们需要在责任链中使用next，而在这里它使用了filter来找到下一个过滤器，从而正常地放行。
     */

    /**
     * 全局过滤器
     *
     * @param exchange
     * @param chain
     * @return
     */
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        // 1.请求日志
        ServerHttpRequest request = exchange.getRequest();
        String path = INTERFACE_HOST + request.getPath().value();
        String method = request.getMethod().toString();
        log.info("请求唯一标识：" + request.getId());
        log.info("请求路径：" + path);
        log.info("请求方法：" + method);
        log.info("请求参数：" + request.getQueryParams());
        String sourceAddress = request.getLocalAddress().getHostString();
        log.info("请求来源地址：" + sourceAddress);
        log.info("请求来源地址：" + request.getRemoteAddress());
        // 拿到响应对象
        ServerHttpResponse response = exchange.getResponse();

        // 2.访问控制 - 黑白名单
        if (!IP_WHITE_LIST.contains(sourceAddress)) {
            // 设置响应状态码为403 Forbidden(禁止访问)
            response.setStatusCode(HttpStatus.FORBIDDEN);
            // 返回处理完成的响应
            return response.setComplete();
        }
        // 3. 用户鉴权（判断ak、sk是否合法）
        // 从请求头获取参数
        HttpHeaders headers = request.getHeaders();
        String accessKey = headers.getFirst("accessKey");
        String nonce = headers.getFirst("nonce");
        String timestamp = headers.getFirst("timestamp");
        String sign = headers.getFirst("sign");
        String body = headers.getFirst("body");
        // todo:实际情况应该是去数据库张查是否分配给用户
        User invokeUser = null;
        try {
            // 调用内部服务，根据访问密钥获取用户信息
            invokeUser = innerUserService.getInvokeUser(accessKey);
        } catch (Exception e) {
            // 捕获异常，记录日志
            log.error("getInvokeUser error", e);
        }
        if (invokeUser == null) {
            // 如果用户信息为空，处理未授权情况并返回响应
            return handleNoAuth(response);
        }
//        if (!accessKey.equals("cammy")) {
//            throw new RuntimeException("无权限");
//        }
        // 直接校验如果随机数大于一万，则抛出异常，并提示"无权限"
        if (Long.parseLong(nonce) > 10000) {
            throw new RuntimeException("无权限");
        }

        // 时间和当前时间不能超过5分钟
        // 首先，获取当前时间的时间戳，以秒为单位
        // System.currentTimeMillis()返回的是以毫秒为单位的时间戳,除以1000后得到当前时间的秒数
        long currentTime = System.currentTimeMillis() / 1000;
        // 定义一个常量FIVE_MINUTES，表示五分钟的时间间隔（乘以60，将分钟转为秒，得到五分钟的时间间隔）
        final Long FIVE_MINUTES = 5 * 60L;
        // 判断当前时间与传入的时间戳是否相差五分钟或以上
        // Long.parseLong(timestamp)将传入的时间戳转换为Long类型
        // 然后计算当前时间与传入时间戳之间的差值（以秒为单位），如果差值大于等于五分钟，则返回true，否则返回false
        if ((currentTime - Long.parseLong(timestamp)) >= FIVE_MINUTES) {
            // 如果时间戳与当前时间相差五分钟或以上，调用handleNoAuth(response)方法，并返回处理完成的响应
            return handleNoAuth(response);
        }

        // 实际情况中是从数据库中查出secretKey
        // 从获取到的用户信息中获取用户的密钥
        String secretKey = invokeUser.getSecretKey();
        // 使用获取到的密钥对请求体进行签名
        String serverSign = SignUtils.genSign(body, secretKey);
        // 检查请求中的签名是否为空，或者是否与服务器生成的签名不一致
        if (sign == null || !sign.equals(serverSign)) {
            // 如果签名为空或者签名不一致，返回处理未授权的响应
            return handleNoAuth(response);
        }

        // 4.请求的模拟接口是否存在？
        // todo:从数据库中查询模拟接口是否存在，以及请求方式是否匹配（还可以校验请求参数）
        // 初始化一个 InterfaceInfo 对象，用于存储查询结果
        InterfaceInfo interfaceInfo = null;
        try {
            interfaceInfo = innerInterfaceInfoService.getInterfaceInfo(path, method);
        } catch (Exception e) {
            // 如果获取接口信息是出现异常，记录错误日志
            log.error("getInterfaceInfo error", e);
        }
        // 检查是否成功获取到接口信息
        if (interfaceInfo == null) {
            // 如果为获取到接口信息，返回处理未授权的响应
            return handleNoAuth(response);
        }
        // 5.请求转发，调用模拟接口
        // Mono<Void> filter = chain.filter(exchange);

        // 6.响应日志
        return handleResponse(exchange, chain, interfaceInfo.getId(), invokeUser.getId());
//        log.info("响应状态码：" + response.getStatusCode());
//        // 7.todo:调用成功，接口调用次数 + 1
//        if (response.getStatusCode() == HttpStatus.OK) {
//
//        } else {
//            // 8.调用失败，返回一个规范的错误码
//            return handleInvokeError(response);
//        }
//         return filter;

    }

    private Mono<Void> handleNoAuth(ServerHttpResponse response) {
        response.setStatusCode(HttpStatus.FORBIDDEN);
        return response.setComplete();
    }

    private Mono<Void> handleInvokeError(ServerHttpResponse response) {
        response.setStatusCode(HttpStatus.INTERNAL_SERVER_ERROR);
        return response.setComplete();
    }

    /**
     * 处理响应
     *
     * @param exchange
     * @param chain
     * @param interfaceInfoId
     * @param userId
     * @return
     */
    private Mono<Void> handleResponse(ServerWebExchange exchange, GatewayFilterChain chain, long interfaceInfoId, long userId) {
        try {
            ServerHttpResponse originalResponse = exchange.getResponse();
            DataBufferFactory bufferFactory = originalResponse.bufferFactory();
            HttpStatus statusCode = originalResponse.getStatusCode();
            if (statusCode != HttpStatus.OK) {
                // 对于非200 OK的请求，进行降级处理，直接返回
                return chain.filter(exchange);
            }
            ServerHttpResponseDecorator decoratedResponse = new ServerHttpResponseDecorator(originalResponse) {
                @Override
                public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
                    // 判断响应体是否是Flux类型
                    if (body instanceof Flux) {
                        Flux<? extends DataBuffer> fluxBody = Flux.from(body);
                        // 返回一个处理后的响应体
                        // （这里就理解为它在拼接字符串，它把缓冲区的数据取出来，一点一点拼接好）
                        return super.writeWith(fluxBody.map(dataBuffer -> {
                            // 读取响应体的内容并转换为字节数组
                            // 7.todo:调用成功，接口调用次数 + 1 invokeCount
                            try {
                                // 调用内部用户接口信息服务，记录接口调用次数
                                innerUserInterfaceInfoService.invokeCount(interfaceInfoId, userId);
                            } catch (Exception e) {
                                // 捕获异常并记录错误信息
                                log.error("invokeCount error", e);
                            }
                            byte[] content = new byte[dataBuffer.readableByteCount()];
                            dataBuffer.read(content);
                            DataBufferUtils.release(dataBuffer);//释放掉内存
                            // 构建返回日志
                            StringBuilder stringBuilder = new StringBuilder(200);
//                            stringBuilder.append("<--- {} {} \n");
                            List<Object> rspArgs = new ArrayList<>();
                            rspArgs.add(originalResponse.getStatusCode());
//                            rspArgs.add(requestUrl);
                            String data = new String(content, StandardCharsets.UTF_8);
                            stringBuilder.append(data);
//                            log.info(stringBuilder.toString(), rspArgs.toArray()); // 打印日志
                            log.info("响应结果：" + data);
                            // 将处理后的内容重新包装成DataBuffer并返回
                            return bufferFactory.wrap(content);
                        }));
                    } else {
                        // 8.调用失败，返回一个规范的错误码
                        log.error("<-- {} 响应code异常", getStatusCode());
                    }
                    return super.writeWith(body); // 降级处理返回数据
                }
            };
            // 对于200 OK的请求，将装饰后的响应对象传递给下一个过滤器链，并继续处理（设置response对象为装饰过的）
            return chain.filter(exchange.mutate().response(decoratedResponse).build());

        } catch (Exception e) {
            // 处理异常情况，记录错误日志
            log.error("网关处理响应异常：" + e);
            return chain.filter(exchange);
        }
    }

    @Override
    public int getOrder() {
        return -1;
    }
}